package com.ywps.vaas

import com.github.mrpowers.spark.fast.tests.DataFrameComparer
import com.ywps.vaas.demoTask.controller.{SparkSqlController, WriteJdbcController}
import com.ywps.vaas.framework.app.ISparkSessionTest
import com.ywps.vaas.framework.util.EnvUtil
import org.scalatest.FunSuite
import org.scalatest.mockito.MockitoSugar

class Demo extends FunSuite with ISparkSessionTest with MockitoSugar
  with DataFrameComparer with Serializable{

  test("testDemo"){
    EnvUtil.put(spark)
    val controller = new SparkSqlController
    controller.execute()
    EnvUtil.remove()
  }

  test("writeJdbcDemo"){
    EnvUtil.put(spark)
    val controller = new WriteJdbcController
    controller.execute()
    EnvUtil.remove()
  }
}
